import os
import hashlib
import zipfile
from common.consts import REPORT_PATH
def md5_code(password):
    md5 = hashlib.md5()
    md5.update(password.encode('utf-8'))
    return md5.hexdigest()

def trans_str(s):
    s=s.replace('":"',"=").replace('":"',"=")
    return s

def zip_dir(startdir,zip_file):
    #压缩文件夹
    _zip_file = os.path.join(REPORT_PATH,zip_file)
    if os.path.exists(_zip_file):
        #先删除
        os.remove(_zip_file)
    z = zipfile.ZipFile(_zip_file, "w", zipfile.ZIP_DEFLATED)
    for dir_path, dir_names, file_names in os.walk(startdir):
        fpath = dir_path.replace(startdir,'')
        #不replace的话，就从根目录开始复制
        fpath = fpath and fpath + os.sep or ''
        for filename in file_names:
            z.write(os.path.join(dir_path, filename),fpath+filename)
    z.close()